package com.qlc_demo_api.util;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTVerifier;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.exceptions.JWTDecodeException;
import com.auth0.jwt.interfaces.DecodedJWT;
import org.apache.shiro.authc.UnknownAccountException;

import java.io.UnsupportedEncodingException;
import java.util.Date;

public class JWTUtil {

    // 24小时 转为毫秒（过期时间）
    private static final long EXPRIRE_TIME = 24 * 60 * 60 * 1000;
//    private static final long EXPRIRE_TIME = 60 * 1000;

    // 密钥
    private static final String SECRET = "zw";

    /**
     * 生成etoken
     * @param username
     * @return
     */
    public static String createToken(String username) {
        try {
            Date date = new Date(System.currentTimeMillis() + EXPRIRE_TIME);
            Algorithm algorithm = Algorithm.HMAC256(SECRET);

            return JWT.create()
                    // 附带username信息
                    .withClaim("username", username)
                    // 到期时间
                    .withExpiresAt(date)
                    // 颁发时间，即该令牌被创建的时间
                    .withIssuedAt(new Date())
                    // 创建一个新的JWT，并使用给定的算法进行标记
                    .sign(algorithm);

        } catch (UnsupportedEncodingException e) {
            return null;
        }
    }

    /**
     * 校验token是否正确
     * @param etoken
     * @param username
     * @return
     */
    public static boolean verify(String etoken, String username) {
        try {
            Algorithm algorithm = Algorithm.HMAC256(SECRET);
            // 在etoken中带了username信息
            JWTVerifier verifier = JWT.require(algorithm).withClaim("username", username).build();

            // 验证token
            verifier.verify(etoken);
            return true;
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 获取etoken信息，无需secrect解密也能获得
     * @param etoken
     * @return
     */
    public static String getUsername(String etoken) {
        try{
            DecodedJWT jwt = JWT.decode(etoken);
            return jwt.getClaim("username").asString();
        }catch (JWTDecodeException e) {
            return null;
        }
    }

    /**
     * 获取token列名
     * @param etoken
     * @param name
     * @return
     */
    public static String getClaim(String etoken, String name) {
        String claim = null;
        try {
            claim = JWT.decode(etoken).getClaim(name).asString();
        }catch (Exception e) {
            return "getClaimFalse";
        }
        return claim;
    }
}
